BeNice!:通过 KV 共享弥合循环架构推理内存差距
🧭 LLM 循环架构 · KV-cache 共享
University of California, San Diego · Together AI
用随机深度预训练加末层KV广播后训练,实现循环模型跨层KV共享且不掉点
📌 工作参考价值(4 点)
- 部署 LLM 作为长用户行为序列编码器或生成式推荐 backbone 时,如果采用 looped/weight-tied Transformer,KV cache 不再随循环次数线性占用显存;BeNice 的末次循环 KV 广播 + 首次循环适配可作为低显存长上下文推理方案,适合 Agent 长会话和超长商品描述场景。
- 两阶段思路可迁移:想要模型获得 KV-shareability,不是直接微调,而是在(继续)预训练中引入 stochastic depth / 随机层丢弃,让残差路径收缩到近似固定点;否则同结构不同初始化可能不具备共享性质。
- 后训练可以保持不变原模型主体,只训练首次 recurrence 对末次 KV 的兼容层(或轻量适配器),避免破坏已收敛表示;这与 MLA、GQA、Kimi delta attention 以及 fp8/nvfp4 等量化方案可叠加,可直接纳入现有推理优化 stack。
- 若当前线上不是 looped 架构,建议先把该特性纳入生成式推荐 / Agent 长上下文模型选型:循环 backbone 的参数效率高,但需配套 KV 共享/压缩能力才能获得端到端低显存收益。
KV²:一种自精炼 KV 缓存压缩方法
🧭 LLM 推理 · KV cache 压缩 ★ 深度解读
Karlsruhe Institute of Technology
用轻量代理选少量锚点做选择性重建,在 2% KV 预算下大幅优于 KVzip 等基线
📌 工作参考价值(4 点)
- 对 Agent 长期记忆、用户历史、商品知识库等“一次缓存、多次 query”场景,优先用 query-agnostic 压缩;KV² 两阶段思路可复制:先用便宜业务信号(如 key 相似度、商品/行为 embedding)挑锚点,再做局部 max-attention 打分,比全量重建省钱。
- 可将 KV 预算压到 2%–5%,适合电商搜索/推荐的超长上下文(长用户序列、多商品文档),降低缓存显存和 decode 带宽;chunk size 4K 是性价比甜点。
- 默认单轮即可,别急着上多轮 self-refinement;只有在 proxy 很弱或 reconstruction 比例极小时才考虑 2–3 轮。
- 落地注意:论文基于 kvpress 的 masking 实现未物理 compact,需配合 headwise 变长 attention、paged cache 和重复 query 服务后端,才能把压缩阶段收益转成实际 decode 收益。
超越 KV Cache:混合语言模型跨模型循环记忆迁移(4B→9B 无前缀重放)
🧭 LLM 跨模型记忆状态迁移
首次展示不同尺寸混合语言模型间直接迁移 Gated DeltaNet 持久循环状态,无需目标模型重新读取上下文,显著降低续写损失
📌 工作参考价值(3 点)
- 在长上下文 LLM 推理服务中,可从 4B 草稿模型迁移注意力 KV 与循环状态到 9B 主模型,避免重算历史前缀,显著降低首 token 延迟;电商 Agent 长会话切换模型时可借鉴「直连复用 recurrent/conv state + 轻量 rank-4 修正」作为状态交接基线,不一定需要训练复杂映射。
- 对生成式推荐/Agent 的长会话记忆:混合模型(full-attention + GDN)的持久状态跨模型直接复用其坐标比学习映射更有效,提示模型家族内 hidden state 有一定兼容性,升级模型版本时可先尝试直接复制状态,再用极少量参数微调纠正。
- 若业务上需做模型蒸馏或小模型生成草稿、大模型验证,可复用该思想:把草稿模型的 KV 和 recurrent state 迁移给大模型,省去大模型重新编码 prompt;rank-4 修正成本极低(约 43 万参数),适合线上快速适配。
前缀滑动:高效测试时扩展
🧭 长程推理 · 前缀滑动窗口 ★ 深度解读
Stanford University · University of California, Santa Barbara · Prime Intellect · University of Washington
保留前缀与最近滑动窗口、丢弃中间推理 token,实现恒定成本的长程推理,无训练 3 倍加速
📌 工作参考价值(4 点)
- 长程 Agent/工具调用(多跳搜索、自动比价、复杂排障):把 system prompt、工具定义、任务描述固定为 prefix,中间推理只保留最近 2048-4096 的 sliding window;能显著降低 KV cache 与时延,且基本不掉点。
- 用 GRPO/在线 RL 训长 reasoning/Agent 模型时,直接复用 truncated backpropagation:只把最后 4×window tokens 传给 trainer,loss 只算最后 window;可避开 OOM,KL 偏差小,适合异步 RL 长 rollout。
- 工程上避免用 last k / summary 做长上下文压缩:会重复处理 token 或引入额外摘要 step,导致显存锯齿与延迟尖峰;Prefix Sliding 的 custom kernel 用 tile skipping 能保持吞吐稳定,接近 vanilla sliding window。
- 注意适用边界:若 Agent 要读长文档、长 API 返回或长代码补全(类似 LiveCodeBench),窗口建议至少 16K,或把关键证据动态追加到 prefix;短文本任务(query 改写、短推荐理由)加速有限。
用循环Transformer弥合隐式与显式推理差距
🧭 隐式CoT推理加速 · 循环Transformer
UW-Madison · Microsoft Research · ETH Zürich · KRAFTON · Ludo Robotics
首次在3B规模用循环Transformer+并行监督弥合隐式与显式思维链性能差距,并降低推理延迟2.5-6.9倍
📌 工作参考价值(4 点)
- 在电商搜索推荐Agent中,可用隐式循环推理替代显式CoT,将多步决策压缩为并行隐状态迭代,显著降低线上延迟;需为每个推理步设计中间监督信号(如意图、子查询),可用已有行为数据或蒸馏构造。
- 循环权重共享在不增加参数的前提下加深推理深度,适合线上部署的资源受限环境,可直接应用于Agent的轻量级推理模块。
- 可解释性验证技巧(将隐状态投影至词表)可用于监控Agent推理过程,确保推荐/搜索决策的合理性与可解释性。
- 若业务存在用户决策路径数据(如浏览-加购-购买),可仿照中间步骤监督,训练隐式推理模型以加速商品推荐理由生成或解释生成。
LESS 即更多:扩散语言模型的互稳定采样
🧭 扩散语言模型推理加速
MBZUAI · Ecole Polytechnique
提出训练无关的自适应采样器 LESS,通过联合稳定性规则在线决定 token 揭示时机,以 72.1% 更少步骤保持精度。
📌 工作参考价值(4 点)
- 若业务中尝试将扩散模型用于生成式推荐(如 Semantic ID 生成),LESS 可嵌入推理流程,自适应提前终止不稳定的 token,直接减少 Transformer 前向次数并降低延迟。
- 其 top-K JS 散度稳定性判据与 top-1 置信度/持久性规则可以独立迁移到其他 token 级早停(early exiting)场景,比如在序列生成中按位置动态分配计算量。
- 该采样器无需训练、模型无关,可作为即插即用模块快速集成到现有扩散 LLM 部署中,无需重新训练或修改模型权重。
- 本文主要学术贡献在于扩散 LM 的高效采样,若业务尚未使用扩散模型,直接可借鉴点有限;但在 Agent 系统里若用扩散模型做规划或工具调用生成,LESS 可加速推理。
仅索引一次:跨层稀疏注意力与共享路由
🧭 长上下文推理 · 稀疏注意力架构
Microsoft Research · Tsinghua University
在 KV 共享架构上复用路由索引,跨层共享 token 级 top-k 选择,以极低开销实现高精度稀疏注意力,大幅提升长上下文推理效率。
📌 工作参考价值(4 点)
- 长上下文推理效率优化:在电商搜索/推荐 Agent 中,需要处理大量历史对话或商品描述时,可借鉴 CLSA 的共享路由思路——使用 YOCO 类架构共享 KV 缓存,并让不同层复用相同的 top-k 路由索引,极大降低 token 稀疏注意力的路由开销,实现接近 block 稀疏的速度却能保持 token 级精度。
- 工程实现 trick:对于已使用 YOCO 或类似 KV 共享的 LLM 服务,只需额外实现一个跨层复用的 indexer,预填充时计算一次 token 级 top-k,解码时各层直接读取该索引进行稀疏注意力,几乎不增加额外成本,即可将长上下文解码吞吐提升数倍。
- Agent 多轮交互场景:当 Agent 需要在长多轮对话中维护完整上下文并频繁解码时,CLSA 能将 KV 缓存存储大幅减少(因子为数倍),并保持解码速度,同时质量几乎无损,适合部署在资源受限的在线服务中。
- 生成式推荐中的长序列生成:若用 LLM 生成个性化推荐理由或连贯的 item 序列,CLSA 可加速长序列的自回归解码,降低延迟,尤其适合对实时性要求高的推荐场景。
用指数衰减记忆增强注意力提升查询感知KV稀疏性
🧭 查询感知KV稀疏 · 指数衰减记忆增强
EPFL · CLAIRE
利用RAT+的指数衰减记忆模块,一致提升Quest、MoBA、SnapKV等查询感知稀疏方法在长上下文任务上的准确性。
📌 工作参考价值(4 点)
- 若电商或Agent系统中部署长上下文LLM(如处理长对话或大量商品描述),可借鉴引入指数衰减记忆模块,继续少量预训练后,配合现有稀疏方法(SnapKV、Quest)大幅减少KV cache内存占用,几乎无损精度。
- 只需追加10B tokens的持续预训练即可让标准模型适应记忆模块,实现即插即用,对资源有限的业务微调较友好。
- 论文提出的两个假设(记忆使注意力头更专注、帮助模型更早定位关键token)可指导设计更适合稀疏推理的架构,例如在注意力头中融合指数衰减偏置。
- 在工程实现上,该记忆模块以递归方式维护KV状态,可与主流稀疏推理框架整合,适合需要高吞吐、低延迟的长上下文服务。